/*-----------------------------------------------------------------------*/
/* Low level disk I/O module SKELETON for FatFs     (C)ChaN, 2019        */
/*-----------------------------------------------------------------------*/
/* If a working storage control module is available, it should be        */
/* attached to the FatFs via a glue function rather than modifying it.   */
/* This is an example of glue functions to attach various exsisting      */
/* storage control modules to the FatFs module with a defined API.       */
/*-----------------------------------------------------------------------*/


#include "ff.h"      /* Obtains integer types */
#include "diskio.h"    /* Declarations of disk functions */

#include <string.h>
#include "flash.h"    /* Example: Header file of existing MMC/SDC contorl module */
#include "sdcard.h"
#include "io.h"

/* Definitions of physical drive number for each drive */
#define DEV_FLASH	0  /* Example: Map Ramdisk to physical drive 0 */
#define DEV_MMC    	1  /* Example: Map MMC/SD card to physical drive 1 */
#define DEV_USB		2  /* Example: Map USB MSD to physical drive 2 */

#if defined ( __ICCARM__ ) /* iar compiler */
  #pragma data_alignment=4
#endif

ALIGNED_HEAD uint8_t flash_data_buffer[512] ALIGNED_TAIL;  /* buf for sd_read_disk/sd_write_disk function used. */

/*-----------------------------------------------------------------------*/
/* Get Drive Status                                                      */
/*-----------------------------------------------------------------------*/
DSTATUS disk_status (
  BYTE pdrv    /* Physical drive nmuber to identify the drive */
)
{
  DSTATUS stat;
  int result;

  switch (pdrv) {
  case DEV_FLASH :
    result = 0;
    stat = (DSTATUS)result;
    return stat;

  case DEV_MMC :
    result = 0;
    stat = (DSTATUS)result;
    return stat;

  case DEV_USB :
    result = 1;
    stat = (DSTATUS)result;
    return stat;
  }
  return STA_NOINIT;
}



/*-----------------------------------------------------------------------*/
/* Inidialize a Drive                                                    */
/*-----------------------------------------------------------------------*/
DSTATUS disk_initialize (
  BYTE pdrv        /* Physical drive nmuber to identify the drive */
)
{
  DSTATUS stat;
  int result;

  switch (pdrv) {
  case DEV_FLASH :
    result = 0;
    stat = (DSTATUS)result;
    return stat;
  case DEV_MMC :
	result = 0;
    stat = (DSTATUS)result;
    return stat;
  case DEV_USB :
    result = 1;
    stat = (DSTATUS)result;
    return stat;
  }
  return STA_NOINIT;
}



/*-----------------------------------------------------------------------*/
/* Read Sector(s)                                                        */
/*-----------------------------------------------------------------------*/
DRESULT disk_read (
  BYTE pdrv,    /* Physical drive nmuber to identify the drive */
  BYTE *buff,    /* Data buffer to store read data */
  LBA_t sector,  /* Start sector in LBA */
  UINT count    /* Number of sectors to read */
)
{
  DRESULT res;
  int result;

  switch (pdrv) {
  case DEV_FLASH :
	  result = 0;
    W25QXX_Read(buff,sector*512,count*512);
    res = (DRESULT)result;
    return res;

  case DEV_MMC :
    result = SD_ReadDisk(buff,sector,count);
    res = (DRESULT)result;
    return res;

  case DEV_USB :
    result = 1;
    res = (DRESULT)result;
    return res;
  }

  return RES_PARERR;
}



/*-----------------------------------------------------------------------*/
/* Write Sector(s)                                                       */
/*-----------------------------------------------------------------------*/
#if FF_FS_READONLY == 0

DRESULT disk_write (
  BYTE pdrv,      /* Physical drive nmuber to identify the drive */
  const BYTE *buff,  /* Data to be written */
  LBA_t sector,    /* Start sector in LBA */
  UINT count      /* Number of sectors to write */
)
{
  DRESULT res;
  int result;

  switch (pdrv) {
  case DEV_FLASH :
	  result = 0;
    W25QXX_Write((uint8_t *)buff,sector*512,count*512);
    res = (DRESULT)result;
    return res;

  case DEV_MMC :
    result = SD_WriteDisk((uint8_t *)buff,sector,count);
    res = (DRESULT)result;
    return res;

  case DEV_USB :
    result = 1;
    res = (DRESULT)result;
    return res;
  }

  return RES_PARERR;
}

#endif


/*-----------------------------------------------------------------------*/
/* Miscellaneous Functions                                               */
/*-----------------------------------------------------------------------*/
DRESULT disk_ioctl (
  BYTE pdrv,    /* Physical drive nmuber (0..) */
  BYTE cmd,    /* Control code */
  void *buff    /* Buffer to send/receive control data */
)
{
  DRESULT res;
  int result;

  switch (pdrv) {
  case DEV_FLASH :
    
	switch(cmd){
      case CTRL_SYNC:
        result = RES_OK;
        break;
      case GET_SECTOR_SIZE:
        *(DWORD*)buff = 512;
        result = RES_OK;
        break;
      case GET_SECTOR_COUNT:
        *(DWORD*)buff = 16 * 1024 * 1024 / 512;
        result = RES_OK;
        break;
      case GET_BLOCK_SIZE:
        *(DWORD*)buff = 512;
        result = RES_OK;
        break;
      default:
        result = RES_PARERR;
        break;
    }
    res = (DRESULT)result;
    return res;
  case DEV_MMC :
    result = 1;
	switch(cmd)
	{
		case CTRL_SYNC:
			SD_CS_L();
			if(SD_WaitReady()==0)result = RES_OK; 
			else result = RES_ERROR;	  
			SD_CS_H();
			break;	 
		case GET_SECTOR_SIZE:
			*(WORD*)buff = 512;
			result = RES_OK;
			break;	 
		case GET_BLOCK_SIZE:
			*(WORD*)buff = 8;
			result = RES_OK;
			break;	 
		case GET_SECTOR_COUNT:
			*(DWORD*)buff = SD_GetSectorCount();
			result = RES_OK;
			break;
		default:
			result = RES_PARERR;
			break;
	}
    res = (DRESULT)result;
    return res;

  case DEV_USB :
    result = 1;
    res = (DRESULT)result;
    return res;
  }

  return RES_PARERR;
}
